Node / ReticulumProjects / MeshChatX.git / files / tests / backend / test_map_manager_extended.py

Displaying Raw • Download

T8b949e# SPDX-License-Identifier: 0BSD

Tff7b72import T7ee787os

Tff7b72import T7ee787shutil

Tff7b72import T7ee787sqlite3

Tff7b72import T7ee787tempfile

Tff7b72from T7ee787unittestT7ee787.T7ee787mock Tff7b72import Te6edf3MagicMockTb4b4b4, Te6edf3patch

Tff7b72import T7ee787pytest

Tff7b72from T7ee787meshchatxT7ee787.T7ee787srcT7ee787.T7ee787backendT7ee787.T7ee787map_manager Tff7b72import Tb4b4b4(

Te6edf3MAX_EXPORT_TILESTb4b4b4,

Te6edf3MapManagerTb4b4b4,

Te6edf3is_mbtiles_filenameTb4b4b4,

Te6edf3is_path_within_dirTb4b4b4,

Tb4b4b4)

Tf0883e@pytestTff7b72.Td2a8fffixture

Tff7b72def Td2a8fftemp_dirTb4b4b4(Tb4b4b4)Tb4b4b4:

Te6edf3dir_path Tff7b72= Te6edf3tempfileTff7b72.Td2a8ffmkdtempTb4b4b4(Tb4b4b4)

Tff7b72yield Te6edf3dir_path

Te6edf3shutilTff7b72.Td2a8ffrmtreeTb4b4b4(Te6edf3dir_pathTb4b4b4)

Tf0883e@pytestTff7b72.Td2a8fffixture

Tff7b72def Td2a8ffmock_configTb4b4b4(Tb4b4b4)Tb4b4b4:

Te6edf3config Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

Te6edf3configTff7b72.Td2a8ffmap_offline_pathTff7b72.Td2a8ffgetTff7b72.Td2a8ffreturn_value Tff7b72= Tff7b72None

Te6edf3configTff7b72.Td2a8ffmap_mbtiles_dirTff7b72.Td2a8ffgetTff7b72.Td2a8ffreturn_value Tff7b72= Tff7b72None

Tff7b72return Te6edf3config

Tff7b72def Td2a8fftest_map_manager_initTb4b4b4(Te6edf3mock_configTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:

Te6edf3mm Tff7b72= Te6edf3MapManagerTb4b4b4(Te6edf3mock_configTb4b4b4, Te6edf3temp_dirTb4b4b4)

Tff7b72assert Te6edf3mmTff7b72.Td2a8ffstorage_dir Tff7b72== Te6edf3temp_dir

Tff7b72def Td2a8fftest_get_offline_path_defaultTb4b4b4(Te6edf3mock_configTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:

Te6edf3mm Tff7b72= Te6edf3MapManagerTb4b4b4(Te6edf3mock_configTb4b4b4, Te6edf3temp_dirTb4b4b4)

Te6edf3default_path Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6ffoffline_map.mbtilesTa5d6ff"Tb4b4b4)

T8b949e# Not exists

Tff7b72assert Te6edf3mmTff7b72.Td2a8ffget_offline_pathTb4b4b4(Tb4b4b4) Tff7b72is Tff7b72None

T8b949e# Exists

Tff7b72with Tffa657openTb4b4b4(Te6edf3default_pathTb4b4b4, Ta5d6ff"Ta5d6ffwTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3fTb4b4b4:

Te6edf3fTff7b72.Td2a8ffwriteTb4b4b4(Ta5d6ff"Ta5d6ffdataTa5d6ff"Tb4b4b4)

Tff7b72assert Te6edf3mmTff7b72.Td2a8ffget_offline_pathTb4b4b4(Tb4b4b4) Tff7b72== Te6edf3default_path

Tff7b72def Td2a8fftest_list_mbtilesTb4b4b4(Te6edf3mock_configTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:

Te6edf3mm Tff7b72= Te6edf3MapManagerTb4b4b4(Te6edf3mock_configTb4b4b4, Te6edf3temp_dirTb4b4b4)

T8b949e# Create some dummy .mbtiles files

Te6edf3f1 Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6ffmap1.mbtilesTa5d6ff"Tb4b4b4)

Te6edf3f2 Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6ffmap2.mbtilesTa5d6ff"Tb4b4b4)

Tff7b72with Tffa657openTb4b4b4(Te6edf3f1Tb4b4b4, Ta5d6ff"Ta5d6ffwTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3fTb4b4b4:

Te6edf3fTff7b72.Td2a8ffwriteTb4b4b4(Ta5d6ff"Ta5d6ff1Ta5d6ff"Tb4b4b4)

Tff7b72with Tffa657openTb4b4b4(Te6edf3f2Tb4b4b4, Ta5d6ff"Ta5d6ffwTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3fTb4b4b4:

Te6edf3fTff7b72.Td2a8ffwriteTb4b4b4(Ta5d6ff"Ta5d6ff22Ta5d6ff"Tb4b4b4)

Te6edf3files Tff7b72= Te6edf3mmTff7b72.Td2a8fflist_mbtilesTb4b4b4(Tb4b4b4)

Tff7b72assert Tffa657lenTb4b4b4(Te6edf3filesTb4b4b4) Tff7b72== T79c0ff2

Tff7b72assert Tffa657anyTb4b4b4(Te6edf3fTb4b4b4[Ta5d6ff"Ta5d6ffnameTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6ffmap1.mbtilesTa5d6ff" Tff7b72for Te6edf3f Tff7b72in Te6edf3filesTb4b4b4)

Tff7b72assert Tffa657anyTb4b4b4(Te6edf3fTb4b4b4[Ta5d6ff"Ta5d6ffsizeTa5d6ff"Tb4b4b4] Tff7b72== T79c0ff2 Tff7b72for Te6edf3f Tff7b72in Te6edf3files Tff7b72if Te6edf3fTb4b4b4[Ta5d6ff"Ta5d6ffnameTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6ffmap2.mbtilesTa5d6ff"Tb4b4b4)

Tff7b72def Td2a8fftest_get_metadataTb4b4b4(Te6edf3mock_configTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:

Te6edf3mm Tff7b72= Te6edf3MapManagerTb4b4b4(Te6edf3mock_configTb4b4b4, Te6edf3temp_dirTb4b4b4)

Te6edf3db_path Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6fftest.mbtilesTa5d6ff"Tb4b4b4)

Te6edf3mock_configTff7b72.Td2a8ffmap_offline_pathTff7b72.Td2a8ffgetTff7b72.Td2a8ffreturn_value Tff7b72= Te6edf3db_path

T8b949e# Create valid sqlite mbtiles

Te6edf3conn Tff7b72= Te6edf3sqlite3Tff7b72.Td2a8ffconnectTb4b4b4(Te6edf3db_pathTb4b4b4)

Te6edf3connTff7b72.Td2a8ffexecuteTb4b4b4(Ta5d6ff"Ta5d6ffCREATE TABLE metadata (name text, value text)Ta5d6ff"Tb4b4b4)

Te6edf3connTff7b72.Td2a8ffexecuteTb4b4b4(Ta5d6ff"Ta5d6ffINSERT INTO metadata VALUES (Ta5d6ff'Ta5d6ffnameTa5d6ff'Ta5d6ff, Ta5d6ff'Ta5d6ffTest MapTa5d6ff'Ta5d6ff)Ta5d6ff"Tb4b4b4)

Te6edf3connTff7b72.Td2a8ffexecuteTb4b4b4(Ta5d6ff"Ta5d6ffINSERT INTO metadata VALUES (Ta5d6ff'Ta5d6ffformatTa5d6ff'Ta5d6ff, Ta5d6ff'Ta5d6ffjpgTa5d6ff'Ta5d6ff)Ta5d6ff"Tb4b4b4)

Te6edf3connTff7b72.Td2a8ffcommitTb4b4b4(Tb4b4b4)

Te6edf3connTff7b72.Td2a8ffcloseTb4b4b4(Tb4b4b4)

Te6edf3metadata Tff7b72= Te6edf3mmTff7b72.Td2a8ffget_metadataTb4b4b4(Tb4b4b4)

Tff7b72assert Te6edf3metadataTb4b4b4[Ta5d6ff"Ta5d6ffnameTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6ffTest MapTa5d6ff"

Tff7b72assert Te6edf3metadataTb4b4b4[Ta5d6ff"Ta5d6ffformatTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6ffjpgTa5d6ff"

Tff7b72def Td2a8fftest_get_tileTb4b4b4(Te6edf3mock_configTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:

Te6edf3mm Tff7b72= Te6edf3MapManagerTb4b4b4(Te6edf3mock_configTb4b4b4, Te6edf3temp_dirTb4b4b4)

Te6edf3db_path Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6fftest.mbtilesTa5d6ff"Tb4b4b4)

Te6edf3mock_configTff7b72.Td2a8ffmap_offline_pathTff7b72.Td2a8ffgetTff7b72.Td2a8ffreturn_value Tff7b72= Te6edf3db_path

Te6edf3conn Tff7b72= Te6edf3sqlite3Tff7b72.Td2a8ffconnectTb4b4b4(Te6edf3db_pathTb4b4b4)

Te6edf3connTff7b72.Td2a8ffexecuteTb4b4b4(

Ta5d6ff"Ta5d6ffCREATE TABLE tiles (zoom_level integer, tile_column integer, tile_row integer, tile_data blob)Ta5d6ff"Tb4b4b4,

Tb4b4b4)

T8b949e# Zoom 0, Tile 0,0. TMS y for 0/0/0 is (1<<0)-1-0 = 0

Te6edf3connTff7b72.Td2a8ffexecuteTb4b4b4(

Ta5d6ff"Ta5d6ffINSERT INTO tiles VALUES (0, 0, 0, ?)Ta5d6ff"Tb4b4b4,

Tb4b4b4(Te6edf3sqlite3Tff7b72.Td2a8ffBinaryTb4b4b4(Ta5d6ffbTa5d6ff"Ta5d6fftile_dataTa5d6ff"Tb4b4b4)Tb4b4b4,Tb4b4b4)Tb4b4b4,

Tb4b4b4)

Te6edf3connTff7b72.Td2a8ffcommitTb4b4b4(Tb4b4b4)

Te6edf3connTff7b72.Td2a8ffcloseTb4b4b4(Tb4b4b4)

Te6edf3tile Tff7b72= Te6edf3mmTff7b72.Td2a8ffget_tileTb4b4b4(T79c0ff0Tb4b4b4, T79c0ff0Tb4b4b4, T79c0ff0Tb4b4b4)

Tff7b72assert Te6edf3tile Tff7b72== Ta5d6ffbTa5d6ff"Ta5d6fftile_dataTa5d6ff"

Tff7b72def Td2a8fftest_start_export_statusTb4b4b4(Te6edf3mock_configTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:

Te6edf3mm Tff7b72= Te6edf3MapManagerTb4b4b4(Te6edf3mock_configTb4b4b4, Te6edf3temp_dirTb4b4b4)

Tff7b72with Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(Te6edf3mmTb4b4b4, Ta5d6ff"Ta5d6ff_run_exportTa5d6ff"Tb4b4b4)Tb4b4b4:

Te6edf3export_id Tff7b72= Te6edf3mmTff7b72.Td2a8ffstart_exportTb4b4b4(Ta5d6ff"Ta5d6fftest_idTa5d6ff"Tb4b4b4, Tb4b4b4[T79c0ff0Tb4b4b4, T79c0ff0Tb4b4b4, T79c0ff1Tb4b4b4, T79c0ff1Tb4b4b4]Tb4b4b4, T79c0ff0Tb4b4b4, T79c0ff1Tb4b4b4)

Tff7b72assert Te6edf3export_id Tff7b72== Ta5d6ff"Ta5d6fftest_idTa5d6ff"

Te6edf3status Tff7b72= Te6edf3mmTff7b72.Td2a8ffget_export_statusTb4b4b4(Te6edf3export_idTb4b4b4)

Tff7b72assert Te6edf3statusTb4b4b4[Ta5d6ff"Ta5d6ffstatusTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6ffstartingTa5d6ff"

Tff7b72def Td2a8fftest_count_export_tiles_world_low_zoomTb4b4b4(Te6edf3mock_configTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:

Te6edf3mm Tff7b72= Te6edf3MapManagerTb4b4b4(Te6edf3mock_configTb4b4b4, Te6edf3temp_dirTb4b4b4)

Te6edf3bbox Tff7b72= Tb4b4b4[Tff7b72-T79c0ff180Tb4b4b4, Tff7b72-T79c0ff85.051129Tb4b4b4, T79c0ff180Tb4b4b4, T79c0ff85.051129Tb4b4b4]

Te6edf3n Tff7b72= Te6edf3mmTff7b72.Td2a8ffcount_export_tilesTb4b4b4(Te6edf3bboxTb4b4b4, T79c0ff0Tb4b4b4, T79c0ff4Tb4b4b4)

Tff7b72assert Te6edf3n Tff7b72> T79c0ff0

Tff7b72assert Te6edf3n Tff7b72< Te6edf3MAX_EXPORT_TILES

Tff7b72def Td2a8fftest_count_export_tiles_dedupesTb4b4b4(Te6edf3mock_configTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:

Te6edf3mm Tff7b72= Te6edf3MapManagerTb4b4b4(Te6edf3mock_configTb4b4b4, Te6edf3temp_dirTb4b4b4)

Te6edf3single Tff7b72= Te6edf3mmTff7b72.Td2a8ffcount_export_tilesTb4b4b4(Tb4b4b4[T79c0ff0Tb4b4b4, T79c0ff0Tb4b4b4, T79c0ff0.0001Tb4b4b4, T79c0ff0.0001Tb4b4b4]Tb4b4b4, T79c0ff2Tb4b4b4, T79c0ff2Tb4b4b4)

Tff7b72assert Te6edf3single Tff7b72> T79c0ff0

Tff7b72def Td2a8fftest_is_path_within_dir_allows_nested_fileTb4b4b4(Te6edf3mock_configTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:

Te6edf3nested Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6ffmapsTa5d6ff"Tb4b4b4, Ta5d6ff"Ta5d6ffarea.mbtilesTa5d6ff"Tb4b4b4)

Te6edf3osTff7b72.Td2a8ffmakedirsTb4b4b4(Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffdirnameTb4b4b4(Te6edf3nestedTb4b4b4)Tb4b4b4, Te6edf3exist_okTff7b72=Tff7b72TrueTb4b4b4)

Tff7b72assert Te6edf3is_path_within_dirTb4b4b4(Te6edf3nestedTb4b4b4, Te6edf3temp_dirTb4b4b4)

Tff7b72def Td2a8fftest_is_path_within_dir_rejects_parent_traversalTb4b4b4(Te6edf3mock_configTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:

Te6edf3outside Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffdirnameTb4b4b4(Te6edf3temp_dirTb4b4b4)Tb4b4b4, Ta5d6ff"Ta5d6ffescape.mbtilesTa5d6ff"Tb4b4b4)

Tff7b72assert Tff7b72not Te6edf3is_path_within_dirTb4b4b4(Te6edf3outsideTb4b4b4, Te6edf3temp_dirTb4b4b4)

Tff7b72def Td2a8fftest_is_path_within_dir_case_insensitiveTb4b4b4(Te6edf3monkeypatchTb4b4b4, Te6edf3mock_configTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:

Tff7b72def Td2a8fffake_normcaseTb4b4b4(Te6edf3pathTb4b4b4)Tb4b4b4:

Tff7b72return Te6edf3pathTff7b72.Td2a8fflowerTb4b4b4(Tb4b4b4)

Te6edf3monkeypatchTff7b72.Td2a8ffsetattrTb4b4b4(Te6edf3osTff7b72.Td2a8ffpathTb4b4b4, Ta5d6ff"Ta5d6ffnormcaseTa5d6ff"Tb4b4b4, Te6edf3fake_normcaseTb4b4b4)

Te6edf3nested Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6ffMAP.MBTILESTa5d6ff"Tb4b4b4)

Tff7b72assert Te6edf3is_path_within_dirTb4b4b4(Te6edf3nestedTb4b4b4, Te6edf3temp_dirTb4b4b4)

Tff7b72def Td2a8fftest_is_mbtiles_filename_case_insensitiveTb4b4b4(Tb4b4b4)Tb4b4b4:

Tff7b72assert Te6edf3is_mbtiles_filenameTb4b4b4(Ta5d6ff"Ta5d6ffworld.MBTILESTa5d6ff"Tb4b4b4)

Tff7b72assert Tff7b72not Te6edf3is_mbtiles_filenameTb4b4b4(Ta5d6ff"Ta5d6ffnotes.txtTa5d6ff"Tb4b4b4)